
"""
自动生成素描草图
这个脚本可以把彩色图片转化为铅笔素描草图，对人像、景色都有很好的效果。

"""
import numpy as np
from PIL import Image
from PIL import ImageDraw
from PIL import ImageFont

# 自动生成素描草图
def drawPic():
    a = np.asarray(Image.open(r"../File/image1.jpg").convert('L')).astype('float')
    depth = 10.  # (0-100)
    grad = np.gradient(a)  # 取图像灰度的梯度值
    grad_x, grad_y = grad  # 分别取横纵图像梯度值
    grad_x = grad_x * depth / 100.
    grad_y = grad_y * depth / 100.
    A = np.sqrt(grad_x ** 2 + grad_y ** 2 + 1.)
    uni_x = grad_x / A
    uni_y = grad_y / A
    uni_z = 1. / A
    vec_el = np.pi / 2.2  # 光源的俯视角度，弧度值
    vec_az = np.pi / 4.  # 光源的方位角度，弧度值
    dx = np.cos(vec_el) * np.cos(vec_az)  # 光源对x 轴的影响
    dy = np.cos(vec_el) * np.sin(vec_az)  # 光源对y 轴的影响
    dz = np.sin(vec_el)  # 光源对z 轴的影响
    b = 255 * (dx * uni_x + dy * uni_y + dz * uni_z)  # 光源归一化
    b = b.clip(0, 255)
    im = Image.fromarray(b.astype('uint8'))  # 重构图像
    im.save(r"../File/image-1.jpg")
    print("保存成功,请查看")


# 添加水印
def add_watermark(watermark_text):
    image = Image.open("../File/image1.jpg")
    draw = ImageDraw.Draw(image)
    font = ImageFont.load_default_imagefont()
    draw.text((10, 10), watermark_text, fill=(255, 255, 255, 128), font=font)
    image.save("../File/image1-water.jpg")


drawPic()
add_watermark("123")